programming prime number in python